How to assign value to variable in list [PYTHON]